package 二叉树;

public class No114二叉树展开为链表 {

    /**
     * 给你二叉树的根结点 root ，请你将它展开为一个单链表：
     * 展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
     * 展开后的单链表应该与二叉树 先序遍历 顺序相同。
     *  
     * 示例 1：
     * 输入：root = [1,2,5,3,4,null,6]
     * 输出：[1,null,2,null,3,null,4,null,5,null,6]
     * 示例 2：
     * 输入：root = []
     * 输出：[]
     * 示例 3：
     * 输入：root = [0]
     * 输出：[0]
     *  
     * 提示：
     * 树中结点数在范围 [0, 2000] 内
     * -100 <= Node.val <= 100
     *  
     * 进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？
     */

    public void flatten(TreeNode root) {
        convertTreeNodeToListNode(root);
    }

    //二叉树转链表函数
    private void convertTreeNodeToListNode(TreeNode node) {

        if(node==null){
            return;
        }

        convertTreeNodeToListNode(node.left);
        convertTreeNodeToListNode(node.right);

        TreeNode tempNode=node.right;

        node.right=node.left;
        TreeNode cur=node.left;
        node.left=null;
        while (cur!=null&&cur.right!=null){
            cur=cur.right;
        }
        if(cur!=null){
            cur.right=tempNode;
        }else{
            node.right=tempNode;
        }

    }

    public static void main(String[] args) {
        TreeNode treeNode =
                TreeNode.getNodeByArr(new Integer[]{1, 2, 3, 4, 5, 6, 7, 8});
        treeNode.printNode();
        No114二叉树展开为链表 n=new No114二叉树展开为链表();
        n.convertTreeNodeToListNode(treeNode);
        treeNode.printNode();
    }

}
